FlywaySetup.java

package de.twenty11.skysail.server.ext.notes.db;

/*
 * Copyright 2013 graefca.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import java.util.Map;

import javax.persistence.EntityManagerFactory;

import org.apache.commons.dbcp.BasicDataSource;

import com.googlecode.flyway.core.Flyway;

/**
 * @author carsten
 * 
 */
public class FlywaySetup {

    // seems like I cannot use comma-separated locations here...
    private static final String DEFAULT_LOCATIONS = "dbmig/server_ext_notes/other";

    private EntityManagerFactory enitityManagerFactory;
    private Flyway flyway;

    public FlywaySetup() {
        flyway = new Flyway();
        flyway.setLocations(DEFAULT_LOCATIONS);
    }

    public void init() {

        Map<?, ?> properties = (Map<?, ?>) enitityManagerFactory.getProperties().get("PUnitInfo");

        BasicDataSource bds = new BasicDataSource();

        if (properties == null) { // test case
            bds.setUrl((String) enitityManagerFactory.getProperties().get("javax.persistence.jdbc.url"));
            bds.setPassword((String) enitityManagerFactory.getProperties().get("javax.persistence.jdbc.password"));
            bds.setUsername((String) enitityManagerFactory.getProperties().get("javax.persistence.jdbc.user"));
            bds.setDriverClassName((String) enitityManagerFactory.getProperties().get("javax.persistence.jdbc.driver"));

        } else {
            bds.setUrl((String) properties.get("driverUrl"));
            bds.setPassword((String) properties.get("driverPassword"));
            bds.setUsername((String) properties.get("driverUser"));
            bds.setDriverClassName((String) properties.get("driverClassName"));
        }

        flyway.setDataSource(bds);
        flyway.setTable("skysail_server_ext_notes_schema_version");
        // flyway.setLocations();

        ClassLoader ccl = Thread.currentThread().getContextClassLoader();
        ClassLoader thisClassLoader = this.getClass().getClassLoader();
        Thread.currentThread().setContextClassLoader(thisClassLoader);

        flyway.setInitOnMigrate(true);

        flyway.migrate();
        Thread.currentThread().setContextClassLoader(ccl);

    }

    public synchronized void setEntityManager(EntityManagerFactory emf) {
        this.enitityManagerFactory = emf;
    }

    public void setLocation(String locations) {
        flyway.setLocations(locations);
    }
}